package org.leo.dm.util.comm;

import java.io.BufferedInputStream;
import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.OutputStreamWriter;
import java.io.Writer;

import org.apache.commons.io.FileUtils;
import org.apache.commons.io.FilenameUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class FileUtil {

	private static final Logger logger = LoggerFactory.getLogger(FileUtil.class);

	/**
	 * 创建目录
	 */
	public static File createDir(String dirPath) {
		File dir;
		try {
			dir = new File(dirPath);
			if (!dir.exists()) {
				FileUtils.forceMkdir(dir);
			}
		} catch (Exception e) {
			logger.error("创建目录出错！", e);
			throw new RuntimeException(e);
		}
		return dir;
	}

	/**
	 * 创建文件
	 */
	public static File createFile(String filePath) {
		File file;
		try {
			file = new File(filePath);
			File parentDir = file.getParentFile();
			if (!parentDir.exists()) {
				FileUtils.forceMkdir(parentDir);
			}
		} catch (Exception e) {
			logger.error("创建文件出错！", e);
			throw new RuntimeException(e);
		}
		return file;
	}

	/**
	 * 复制目录（不会复制空目录）
	 */
	public static void copyDir(String srcPath, String destPath) {
		try {
			File srcDir = new File(srcPath);
			File destDir = new File(destPath);
			if (srcDir.exists() && srcDir.isDirectory()) {
				FileUtils.copyDirectoryToDirectory(srcDir, destDir);
			}
		} catch (Exception e) {
			logger.error("复制目录出错！", e);
			throw new RuntimeException(e);
		}
	}

	/**
	 * 复制文件
	 */
	public static void copyFile(String srcPath, String destPath) {
		try {
			File srcFile = new File(srcPath);
			File destDir = new File(destPath);
			if (srcFile.exists() && srcFile.isFile()) {
				FileUtils.copyFileToDirectory(srcFile, destDir);
			}
		} catch (Exception e) {
			logger.error("复制文件出错！", e);
			throw new RuntimeException(e);
		}
	}

	/**
	 * 删除目录
	 */
	public static void deleteDir(String dirPath) {
		try {
			File dir = new File(dirPath);
			if (dir.exists() && dir.isDirectory()) {
				FileUtils.deleteDirectory(dir);
			}
		} catch (Exception e) {
			logger.error("删除目录出错！", e);
			throw new RuntimeException(e);
		}
	}

	/**
	 * 删除文件
	 */
	public static void deleteFile(String filePath) {
		try {
			File file = new File(filePath);
			if (file.exists() && file.isFile()) {
				FileUtils.forceDelete(file);
			}
		} catch (Exception e) {
			logger.error("删除文件出错！", e);
			throw new RuntimeException(e);
		}
	}

	/**
	 * 重命名文件
	 */
	public static void renameFile(String srcPath, String destPath) {
		File srcFile = new File(srcPath);
		if (srcFile.exists()) {
			File newFile = new File(destPath);
			boolean result = srcFile.renameTo(newFile);
			if (!result) {
				throw new RuntimeException("重命名文件出错！" + newFile);
			}
		}
	}

//	/**
//	 * 读取文件
//	 */
//	public byte[] readFile(String filename) throws IOException {
//		BufferedInputStream bufferedInputStream = new BufferedInputStream(
//				getClass().getResource(filename).openStream());
//		int len = bufferedInputStream.available();
//		byte[] bytes = new byte[len];
//		int r = bufferedInputStream.read(bytes);
//		if (len != r) {
//			bytes = (byte[]) null;
//			throw new IOException("读取文件不正确");
//		}
//		bufferedInputStream.close();
//		return bytes;
//	}

	/**
	 * 将字符串写入文件
	 */
	public static void writeFile(String filePath, String fileContent) {
		OutputStream os = null;
		Writer w = null;
		try {
			FileUtil.createFile(filePath);
			os = new BufferedOutputStream(new FileOutputStream(filePath));
			w = new OutputStreamWriter(os, "UTF_8");
			w.write(fileContent);
			w.flush();
		} catch (Exception e) {
			logger.error("写入文件出错！", e);
			throw new RuntimeException(e);
		} finally {
			try {
				if (os != null) {
					os.close();
				}
				if (w != null) {
					w.close();
				}
			} catch (Exception e) {
				logger.error("释放资源出错！", e);
			}
		}
	}

	/**
	 * 获取真实文件名（去掉文件路径）
	 */
	public static String getRealFileName(String fileName) {
		return FilenameUtils.getName(fileName);
	}

	/**
	 * 判断文件是否存在
	 */
	public static boolean checkFileExists(String filePath) {
		return new File(filePath).exists();
	}
	
	public static byte[] readFileJar(String filename)
		    throws IOException
		  {
			InputStream in = FileUtil.class.getResourceAsStream(filename);
		    BufferedInputStream bufferedInputStream = new BufferedInputStream(in);
		    int len = in.available();
		    byte[] bytes = new byte[len];
		    int r = bufferedInputStream.read(bytes);
		    if (len != r)
		    {
		      bytes = (byte[])null;
		      throw new IOException("读取文件不正确");
		    }
		    bufferedInputStream.close();
		    return bytes;
		  }
	
	public static byte[] readFile(String filename) throws IOException {
		File file = new File(filename);
		if ((filename == null) || (filename.equals(""))) {
			throw new NullPointerException("无效的文件路径");
		}
		long len = file.length();
		byte[] bytes = new byte[(int) len];

		BufferedInputStream bufferedInputStream = new BufferedInputStream(new FileInputStream(file));
		int r = bufferedInputStream.read(bytes);
		if (r != len) {
			throw new IOException("读取文件不正确");
		}
		bufferedInputStream.close();

		return bytes;
	}
	
}
